CUDA_VISIBLE_DEVICES=1 python3 main.py --mode test --model res_cbam